
% compute_energy.m ——— 计算忆阻神经元模型的瞬时能量 H
% 用法：
%   H = compute_energy(X, alpha, a, b)
% 输入：
%   X     — 3×M 状态矩阵，每一列是 [x; y; z]
%   alpha — 无量纲参数 α
%   a     — 无量纲参数 a′
%   b     — 无量纲参数 b′
% 输出：
%   H     — 1×M 向量，对应每个时刻的能量值

function H = energy(X, alpha, a, b)
    % X(1,:) = x, X(2,:) = y, X(3,:) = z
    x = X(1,:);
    y = X(2,:);
    z = X(3,:);
    % 按论文式 (6): H = 0.5*α x^2 + 0.5 y^2 + 0.5 a z x + 1.5 b z^3 x
    H = 0.5*alpha .* (x.^2) ...
      + 0.5          .* (y.^2) ...
      + 0.5*a        .* (z .* x) ...
      + 1.5*b        .* (z.^3 .* x);
end